Pytorch中torch.full(),torch.ones()和torch.zeros()函数解析 |
您所在的位置:网站首页 › pytorch new_ones › Pytorch中torch.full(),torch.ones()和torch.zeros()函数解析 |
一. torch.full()函数解析
1. 官网链接
torch.full(),如下图所示: 返回创建size大小的维度,里面元素全部填充为fill_value 3.代码举例输出形状为(2,3)的二维张量,里面元素全部填充为5 x = torch.full(size=(2,3),fill_value=5) x 输出结果如下: tensor([[5, 5, 5], [5, 5, 5]]) 二. torch.ones()函数解析 1. 官网链接torch.ones(),如下图所示: 返回创建size大小的维度,里面元素全部填充为1 3. 代码举例 a = torch.ones(2, 3) b = torch.ones(5) c = torch.ones(size=(3,4)) a,b,c 输出结果如下: (tensor([[1., 1., 1.], [1., 1., 1.]]), tensor([1., 1., 1., 1., 1.]), tensor([[1., 1., 1., 1.], [1., 1., 1., 1.], [1., 1., 1., 1.]])) 三. torch.zeros()函数解析 1.官网链接torch.zeros(),如下图所示: 返回创建size大小的维度,里面元素全部填充为0 3. 代码举例 a = torch.zeros(2, 3) b = torch.zeros(5) c = torch.zeros(size=(3,4)) a,b,c 输出结果如下: (tensor([[0., 0., 0.], [0., 0., 0.]]), tensor([0., 0., 0., 0., 0.]), tensor([[0., 0., 0., 0.], [0., 0., 0., 0.], [0., 0., 0., 0.]])) |
CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |